import os
import sys
import json
from openai import OpenAI

# 添加项目根目录到Python路径，以便导入config_reader模块
current_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.dirname(os.path.dirname(current_dir))
sys.path.append(project_root)

from config_reader import get_deepseek_api_key

# 从配置文件获取DeepSeek API密钥
api_key = get_deepseek_api_key()
client = OpenAI(api_key=api_key, base_url="https://api.deepseek.com")

# 最终版 Prompt
prompt_v3 = """
生成手机广告，包含：
- 核心卖点(不超过3个)
- 使用场景描述
- 促销信息
输出为JSON格式：
{"slogan": "", "features": [], "promotion": ""}
语言风格：年轻化网络用语，如'YYDS'、'种草'等
"""

try:
    response = client.chat.completions.create(
        model="deepseek-chat",  # 使用 DeepSeek V3 模型
        messages=[
            {"role": "user", "content": prompt_v3}
        ],
        temperature=0.7,  # 适度随机，保证创意
        max_tokens=300
    )

    # 提取内容
    content = response.choices[0].message.content.strip()

    # 解析 JSON
    ad_json = json.loads(content)

    # 美化输出
    print("📱 生成的手机广告文案：")
    print(json.dumps(ad_json, ensure_ascii=False, indent=2))

except Exception as e:
    print("❌ 调用出错：", e)